from .init_model import get_wrapper